#include "Tcpserve.h"
#include "Calculator.h"
#include "protocol.h"
#include <memory>
#include "Daemon.h"
using namespace LogMudule;

using cal_fun = std::function<Response(const Request &req)>;

class Parse
{
public:
    Parse(cal_fun c) : _cal(c)
    {
    }
    std::string Entry(std::string &package)
    {
        // 1. 判断报文的完整性！
        std::string message;
        std::string respstr;
        while (Decode(package, &message))
        {
            LOG(LogLevel::DEBUG) << "Content: \n"
                                 << message;
            if (message.empty())
                break;

            // 2. 反序列化， message是一个曾经被序列化的request
            Request req;
            if (!req.Deserialize(message))
                break;

            std::cout << "#############" << std::endl;
            std::cout << "#############" << std::endl;

            // 3. 计算
            Response resp = _cal(req);

            // 4. 序列化
            std::string res;
            resp.Serialize(res);
            LOG(LogLevel::DEBUG) << "序列化: \n"
                                 << res;

            // 5. 添加长度报头字段！
            Encode(res);

            LOG(LogLevel::DEBUG) << "Encode: \n"
                                 << res;

            // 6. 拼接应答
            respstr += res;
        }
        LOG(LogLevel::DEBUG) << "respstr: \n"
                             << respstr;
        return respstr;
    }

private:
    cal_fun _cal;
};
int main()
{
    // ENABLE_CONSOLE_LOG();
    Daemon(false, false);
    ENABLE_FILE_LOG();
    Calculator mycal;
    Parse myparse([&mycal](const Request &req)
                  { return mycal.Execute(req); });
    std::unique_ptr<Tcpserve> tser = std::make_unique<Tcpserve>([&myparse](std::string &package)
                                                                { return myparse.Entry(package); });
    tser->init();
    tser->start();
    return 0;
}